package top.jolyoulu.core.rdd.operator.transform

import org.apache.spark.rdd.RDD
import org.apache.spark.{SparkConf, SparkContext}

/**
 * @Author: JolyouLu
 * @Date: 2024/2/7 20:51
 * @Description
 */
object Spark08_RDD_Operator_Sample {
  def main(args: Array[String]): Unit = {
    //准备环境 [*]:表示使用当前系统最大核
    val sparkConf: SparkConf = new SparkConf().setMaster("local[*]").setAppName("RDD")
    val sc: SparkContext = new SparkContext(sparkConf)

    val rdd: RDD[Int] = sc.makeRDD(List(1,2,3,4,5,6,7,8,9,10))
    //抽取一个数据
    //参数1：抽取后是否把数据放回，true放回 false丢弃
    //参数2：每条数据可能被抽取的概率，基准值达到这个基准值就算抽中
    //参数3：抽取数据时随机算法的种子，如果种子固定那么抽取的数据就是固定了，不传时种子是随机的
    println(rdd.sample(
      false,
      0.4,
      1
    ).collect().mkString(","))
    //关闭环境
    sc.stop()
  }

}
